python equivalent of c map